from torch_geometric.datasets import Planetoid
dataset = Planetoid(root='/root/shared-nvme/new_code/data', name='Cora')
print(type(dataset))